package com.zjj.algorithm.learning.atguigu.recursion;

/**
 *
 * 八皇后问题@author zjj_admin
 * @date 2022/11/12 14:52
 */
public class EightQueens {

    /**
     * 使用一个以为数组表示皇后的站位，下标表示第 i + 1 个皇后，值表示站在第 val + 1 列
     * 例如 arr = {0,1,2,3,4,5,6,7} ，arr[0] = 0，表示第一个皇后站在 第 1 列
     */
    static int max = 8;
    static int[] array = new int[max];
    static int count = 0;
    static int judgeCount = 0;


    public static void main(String[] args) {

        check(0);
        System.out.println("count = " + count);
        System.out.println("judgeCount = " + judgeCount);

    }

    /**
     * 皇后站位
     * @param n 开始方第 n  个皇后
     */
    public static void check(int n){
        if(max == n){
            count ++;
            print();
            return;
        }
        for (int i = 0; i < max; i++) {
            //现将当前这个皇后 n，放到该行的第一列
            array[n] = i;
            if(judge(n)){
                // 当放入第 n 个皇后之后不冲突时，就放入第 n + 1 个
                check(n + 1);
            }
        }
    }


    /**
     * 判断是否冲突
     * @param n
     * @return
     */
    public static boolean judge(int n){
        judgeCount ++ ;
        for (int i = 0; i < n; i++) {
            if(array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])){
                return false;
            }
        }
        return true;
    }


    /**
     * 打印数组
     */
    public static void print(){
        for (int i = 0; i < array.length - 1; i++) {
            System.out.print(array[i] + " ");
        }
        System.out.println();
    }
}
